'''
大模型和工具绑定
langgraph.prebuilt import create_react_agent
1.创建工具
2.创建llm
3.创建预构建agent绑定模型和工具
4.运行  参数需要messages为key得字典
'''

from model_utils import getLLM
from typing import Annotated
from langchain.tools import tool
from langgraph.prebuilt import create_react_agent


_llm = getLLM()

@tool
def weather(city :Annotated[str, "被查询的城市，用中文输入"]):
    """用于查询输入城市今日的天气情况"""
    if(city=="上海"):
        return "上海今日有台风12级"
    return "天气晴朗，风和日丽"

agent = create_react_agent(_llm,[weather])

from langchain_core.messages import HumanMessage
r = agent.invoke({"messages":[HumanMessage("上海今天天气怎么样?")]})
print(r["messages"][-1].content)